package com.shexianyu.medium;

import java.util.HashMap;
import java.util.Map;

public class MapSum {
    Map<String, Integer> map;

    public MapSum() {
        map = new HashMap<String, Integer>();
    }

    public void insert(String key, int val) {
        map.put(key, val);
    }

    public int sum(String prefix) {
        int res = 0;
        for (String str : map.keySet()) {
            if (str.startsWith(prefix)) {
                res += map.get(str);
            }
        }
        return res;
    }
}
